import qrcode
import random

class Utils:  # 创建一个类，用于操作
    # 去除@用户名
    @staticmethod
    def remove_at(msg):
        # 判断msg是否有self这个属性
        try:
            return msg.text.replace(f"@{msg.sender.self.name}", "").strip()
        except:
            return msg.text.strip()

    # 生成范围的随机数
    @staticmethod
    def random_num(start, end):
        return random.randint(start, end)



    # 文字转换为二维码
    @staticmethod
    def text_to_qrcode(Strs, filename):
        qr = qrcode.QRCode(
            version=5,
            error_correction=qrcode.constants.ERROR_CORRECT_L,
            box_size=10,
            border=4,
        )
        qr.add_data(Strs)
        qr.make(fit=True)
        imgSrc = './images/'+filename + '.png'
        img = qr.make_image(fill_color='pink', back_color='white')
        img = img.convert("RGBA")
        img.save(imgSrc)
        return imgSrc